package com.waimai.app.qa.webdriver;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;


public class AppDriverPlus {

	public static final String PFILEPATH = "/test-output/screenshot";
	
	/**删除文件
	 * @author AiHui
	 * @param file
	 */
	public static void deleteFile(File file) {
		if (file.exists()) {
			if (file.isFile()) {
				file.delete();
			} else if (file.isDirectory()) {
				File files[] = file.listFiles();
				for (int i = 0; i < files.length; i++) {
					deleteFile(files[i]);
				}
			}
		file.delete();
		}
	}
	
	/**
	 * 拷贝文件操作
	 */
	public static void copyFile(File sourceFile,File targetFile)throws IOException{  
        //新建文件输入流并对它进行缓冲   
        FileInputStream input = new FileInputStream(sourceFile);  
        BufferedInputStream inBuff=new BufferedInputStream(input);  
  
        //新建文件输出流并对它进行缓冲   
        FileOutputStream output = new FileOutputStream(targetFile);  
        BufferedOutputStream outBuff=new BufferedOutputStream(output);  
          
        //缓冲数组   
        byte[] b = new byte[1024 * 5];  
        int len;  
        while ((len =inBuff.read(b)) != -1) {  
            outBuff.write(b, 0, len);  
        }  
        //刷新此缓冲的输出流   
        outBuff.flush();  
          
        //关闭流   
        inBuff.close();  
        outBuff.close();  
        output.close();  
        input.close();  
    }  

	/**
	 * 复制目录
	 */
	public static void copyDirectiory(String sourceDir, String targetDir) throws IOException {  
		//新建目标目录   
		(new File(targetDir)).mkdirs();  
		
		//获取源文件夹当前下的文件或目录   
		File[] file = (new File(sourceDir)).listFiles();  
		for (int i = 0; i < file.length; i++) {  
			if (file[i].isFile()) {  
				//源文件   
				File sourceFile=file[i];  
				//目标文件   
				File targetFile=new  File(new File(targetDir).getAbsolutePath()+File.separator+file[i].getName());  
				copyFile(sourceFile,targetFile);  
			}  
			if (file[i].isDirectory()) {  
				//准备复制的源文件夹   
				String dir1=sourceDir + "/" + file[i].getName();  
				//准备复制的目标文件夹   
				String dir2=targetDir + "/"+ file[i].getName();  
				copyDirectiory(dir1, dir2);  
			}  
		}  
	}  
           
	/**
	 * 录屏结束后将文件从源拷贝到目的地
	 */
	public static void sourCopyDir(String sourPath, String dirPath) throws IOException{
		File file1=new File(sourPath);
		File[] file=(file1).listFiles();
   
		if(file != null){
			for(int i=0;i<file.length;i++){
				if(file[i].isFile()){
					//复制文件
					copyFile(file[i], new File(dirPath+file[i].getName()));		   
				}
				if(file[i].isDirectory()){
					String sourceDir=sourPath+File.separator+file[i].getName();	   
					String targetDir=dirPath+File.separator+file[i].getName();
					copyDirectiory(sourceDir, targetDir);       	            	  
				}
			}
			deleteFile(file1);
		}
	}  
}
